home *** CD-ROM | disk | FTP | other *** search
/ Workbench Add-On / Workbench Add-On - Volume 1.iso / BBS-Archive / Comm / AmiTCP30b2.lha / src / appl / qwriter / inet.c < prev    next >
C/C++ Source or Header  |  1993-04-13  |  4KB  |  187 lines
  1. /*
  2.  * Opens requred sockets
  3.  * $Header: inet.c,v 2.1 93/04/13 21:46:08 jraja Exp $
  4.  * $Log:    inet.c,v $
  5.  * Revision 2.1  93/04/13  21:46:08  21:46:08  jraja (Jarno Tapio Rajahalme)
  6.  * Made this compile with the newest API.
  7.  * 
  8.  * Revision 2.0  93/03/20  17:31:39  17:31:39  ppessi (Pekka Pessi)
  9.  * initial netlib version..
  10.  * 
  11.  * Revision 1.4  93/03/16  19:13:57  19:13:57  too (Tomi Ollila)
  12.  * code fixes
  13.  * 
  14.  * Revision 1.3  93/03/16  10:43:25  10:43:25  puhuri (Markus Peuhkuri)
  15.  * Added AMITCP stuff.
  16.  * 
  17.  */
  18.  
  19. #ifdef AMIGA
  20. #include <exec/types.h>
  21. #include <sys/types.h>
  22. #if __SASC
  23. #include <proto/socket.h>
  24. #elif __GNUC__
  25. #include <inline/socket.h>
  26. #else
  27. #include <clib/socket_protos.h>
  28. #endif
  29. #endif /* AMIGA */
  30.  
  31. #ifdef __STDC__
  32. #include <stdlib.h>
  33. #endif
  34.  
  35. #include <stdio.h>
  36.  
  37. #include <sys/types.h>
  38. #include <sys/socket.h>
  39. #include <sys/ioctl.h>
  40. #include <sys/time.h>
  41.  
  42. #include <netinet/in.h>
  43.  
  44. #include <netdb.h>
  45.  
  46. #include "qwriter.h"
  47. #include <string.h>
  48.  
  49. #include <arpa/inet.h>
  50.  
  51. /*
  52.  * hostname pointer to our host name or NULL
  53.  * port in a port number to listen
  54.  * type is SOCK_STREAM (TCP) or SOCK_DGRAM (UDP)
  55.  * 
  56.  * returns a file descriptor to open socket
  57.  */
  58.  
  59. char *MyHost;
  60.  
  61. int Fail(int s, int error)
  62. {
  63.   if (s >= 0)
  64.     CloseSocket(s);
  65.   return error;
  66. }
  67.  
  68. int open_server(char *hostname, UWORD port, int type)
  69. { 
  70. #ifndef NODB
  71.   struct hostent *hp;
  72. #endif
  73.   struct sockaddr_in soc_in_s, soc_in_c;
  74.   int soc_len = sizeof(struct sockaddr);
  75.   int sid_s, sid_c;
  76.  
  77.   printf("'%s' port=%ld type=%ld\n", hostname, port, type);
  78. #ifndef NODB
  79.   if(hostname == NULL){
  80.     hostname = (char *)malloc(80);
  81.     gethostname(hostname,80);
  82.   }
  83.                 /* Get our address */
  84.   if ((hp = gethostbyname(hostname)) == NULL){
  85.     fprintf(stderr,"Couldn't get information for host '%s'",hostname);
  86.     return(FAIL);
  87.   }
  88. #endif
  89.                 /* Get socket */
  90.   if ((sid_s = socket(AF_INET, type, 0)) < 0) {
  91.     perror("socket() failed");
  92.     return(FAIL);
  93.   }
  94.                 /* Create address corresponding our service */
  95.  
  96.   bzero((caddr_t)&soc_in_s, sizeof(soc_in_s));
  97.   bzero((caddr_t)&soc_in_c, sizeof(soc_in_c));
  98.  
  99. #ifdef AMITCP
  100.   soc_in_s.sin_len = sizeof(struct sockaddr_in);
  101. #endif
  102.   soc_in_s.sin_family = AF_INET;
  103.   soc_in_s.sin_port = htons(port);
  104. #ifndef NODB
  105.   bcopy(hp->h_addr, (caddr_t)&soc_in_s.sin_addr, hp->h_length);
  106.   printf("addr: %s\n", inet_ntoa(soc_in_s.sin_addr));
  107. #else
  108.   soc_in_s.sin_addr.s_addr = inet_addr(MyHost);
  109. #endif
  110.  
  111.                 /* Bind it to our socket */
  112.   if(bind(sid_s, (struct sockaddr *)&soc_in_s, sizeof(soc_in_s)) < 0 ) {
  113.     perror("bind() failed");
  114.     return (Fail(sid_s, FAIL));
  115.   }
  116.                 /* An UDP-socket can't be
  117.                    listen'd or accept'd */
  118.   if(type == SOCK_DGRAM)
  119.     return (sid_s);
  120.                 /* Listen socket.. */
  121.   if(listen(sid_s, 1) < 0) {
  122.     perror("listen() failed\n");
  123.     return (Fail(sid_s, FAIL));
  124.   }
  125.                 /* Accept connection */
  126.   if ((sid_c = accept(sid_s, 
  127.               (struct sockaddr *)&soc_in_c, 
  128.               (long *)&soc_len)) 
  129.       < 0){
  130.     perror("accept() failed\n");
  131.     return (Fail(sid_s, FAIL));
  132.   }
  133.  
  134.   CloseSocket(sid_s);
  135.  
  136.   return (sid_c);
  137. }
  138.  
  139. /*
  140.  * as above, but does not accept null as hostname
  141.  */
  142. int open_client(char *hostname,UWORD port,int type)
  143. {
  144. #ifndef NODB
  145.   struct hostent *hp;
  146. #endif
  147.   struct sockaddr_in soc_in_s, soc_in_c;
  148.   int sid_s;
  149.  
  150. printf("'%s' port=%ld type=%ld\n",hostname,port,type);
  151.                 /* Get server address */
  152. #ifndef NODB
  153.   if((hp = gethostbyname(hostname)) == NULL){
  154.     fprintf(stderr,"Couldn't get information for host '%s'",hostname);
  155.     perror("");
  156.     return(FAIL);
  157.   }
  158. #endif
  159.                 /* Get socket */
  160.   if((sid_s = socket(AF_INET, type, 0)) < 0) {
  161.     perror("socket() failed");
  162.     return(FAIL);
  163.   }
  164.  
  165.   bzero((caddr_t)&soc_in_s, sizeof(soc_in_s));
  166.   bzero((caddr_t)&soc_in_c, sizeof(soc_in_c));
  167.  
  168. #ifdef AMITCP
  169.   soc_in_s.sin_len = sizeof(struct sockaddr_in);
  170. #endif
  171.   soc_in_s.sin_family = AF_INET;
  172.   soc_in_s.sin_port = htons(port);
  173. #ifndef NODB
  174.   bcopy(hp->h_addr, (caddr_t)&soc_in_s.sin_addr, hp->h_length);
  175. #else
  176.   soc_in_s.sin_addr.s_addr = inet_addr(MyHost);
  177. #endif
  178.  
  179.                 /* Connect to server.. */
  180.   if (connect(sid_s, (struct sockaddr *)&soc_in_s, sizeof(soc_in_s)) < 0){
  181.     perror("connect() failed");
  182.     CloseSocket(sid_s);
  183.     return(FAIL);
  184.   }
  185.   return(sid_s);
  186. }
  187.